from pathlib import Path
from trend.charts.barchart import TopInstitutesGenerator

def generate_example_institutes():
    """生成示例机构专利数量图"""
    
    # 示例数据
    example_data = {
        "华为技术有限公司": 2500,
        "腾讯科技(深圳)有限公司": 1800,
        "阿里巴巴(中国)有限公司": 1600,
        "百度在线网络技术(北京)有限公司": 1400,
        "字节跳动有限公司": 1200,
        "小米科技有限责任公司": 1000,
        "京东科技有限公司": 800,
        "网易(杭州)网络有限公司": 600,
        "美团点评科技有限公司": 500,
        "滴滴出行科技有限公司": 400
    }
    
    # 创建输出目录
    output_dir = Path(__file__).parent.parent / 'output' / 'barchart'
    output_dir.mkdir(parents=True, exist_ok=True)
    
    # 创建生成器
    generator = TopInstitutesGenerator()
    
    try:
        # 生成柱状图
        save_path = output_dir / 'top_institutes_2023.html'
        result_path = generator.generate(
            data=example_data,
            save_path=str(save_path),
            title='AI领域专利申请机构TOP10'
        )
        print(f"生成成功: {result_path}")
        
    except Exception as e:
        print(f"生成失败: {str(e)}")

if __name__ == "__main__":
    generate_example_institutes() 